# 导入requests模块
import requests
import config


# 创建API类

class CourseApi:

    def __init__(self):
        # 查询所有课程的URL
        self.url_query_all_course = config.BASE_URL + "/ssm_web/course/findAllCourse"
        # 新增课程的URL
        self.url_add_course = config.BASE_URL + "/ssm_web/course/saveOrUpdateCourse"
        # 更新课程状态的URL
        self.url_update_course_status = config.BASE_URL + "/ssm_web/course/updateCourseStatus"

    # 封装查询所有课程接口
    def query_all_course(self, headers):
        return requests.post(url=self.url_query_all_course,
                             json={},
                             headers=headers)

    # 封装新增课程接口
    def add_course(self, headers, request_body):
        return requests.post(url=self.url_add_course,
                             json=request_body,
                             headers=headers)

    # 封装修改课程接口(复用的是添加课程的URL，因为他们是一样的)
    def modify_course(self, headers, request_body):
        return requests.post(url=self.url_add_course,
                             json=request_body,
                             headers=headers)

    # 封装更新课程状态（上架/下架）接口
    def update_course_status(self, headers, status, id):
        return requests.get(url=self.url_update_course_status,
                            headers=headers,
                            params={"status": status, "id": id})
